Skip to content

Refine assert IPEX #17

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
May 20, 2020
Merged

Refine assert IPEX #17

merged 2 commits into from
May 20, 2020

Conversation

EikanWang
Copy link
Contributor

TORCH_INTERNAL_ASSERT is not only for debug but also for release. I replaced it by TORCH_INTERNAL_ASSERT_DEBUG_ONLY. TORCH_INTERNAL_ASSERT_DEBUG_ONLY is only enabled in debug.

@EikanWang
Copy link
Contributor Author

1 similar comment
@EikanWang
Copy link
Contributor Author

@@ -305,11 +305,11 @@ inline std::vector<at::Tensor> _interaction_backward(const at::Tensor & grad_out

at::Tensor AtenIpexTypeExt::interaction_forward(const std::vector<at::Tensor> & input) {
if (input[0].scalar_type() == at::kFloat) {
for (const auto &in : input) { TORCH_INTERNAL_ASSERT(in.scalar_type() == at::kFloat); }
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@hongzhen1 , Do you think it should be replaced by TORCH_CHECK?

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

it's OK for me

@EikanWang EikanWang merged commit a51b709 into intel:master May 20, 2020
zhuhaozhe pushed a commit to zhuhaozhe/intel-extension-for-pytorch that referenced this pull request Apr 23, 2021
add dlrm op bench (using pytorch op bench)
EikanWang pushed a commit that referenced this pull request Oct 4, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants